﻿using EnvDTE;
using EnvDTE80;
using Microsoft.VisualStudio.Shell;
using System.ComponentModel.Design;
using System.Diagnostics;
using System.Globalization;
using System.Runtime.InteropServices;
using System.Windows.Forms;
using System;

namespace VsExt_CopyFileName
{
    /// <summary>
    /// This is the class that implements the package exposed by this assembly.
    ///
    /// The minimum requirement for a class to be considered a valid package for Visual Studio
    /// is to implement the IVsPackage interface and register itself with the shell.
    /// This package uses the helper classes defined inside the Managed Package Framework (MPF)
    /// to do it: it derives from the Package class that provides the implementation of the 
    /// IVsPackage interface and uses the registration attributes defined in the framework to 
    /// register itself and its components with the shell.
    /// </summary>
    // This attribute tells the PkgDef creation utility (CreatePkgDef.exe) that this class is
    // a package.
    // This attribute is used to register the information needed to show the this package
    // in the Help/About dialog of Visual Studio.
    // This attribute is needed to let the shell know that this package exposes some menus.
    [PackageRegistration(UseManagedResourcesOnly = true)]
    [InstalledProductRegistration("#110", "#112", "1.0", IconResourceID = 400)]
    [ProvideMenuResource("Menus.ctmenu", 1)]
    [Guid(GuidList.guidVsExt_CopyFileNamePkgString)]
    public sealed class VsExt_CopyFileNamePackage : Package
    {
        /// <summary>
        /// Default constructor of the package.
        /// Inside this method you can place any initialization code that does not require 
        /// any Visual Studio service because at this point the package object is created but 
        /// not sited yet inside Visual Studio environment. The place to do all the other 
        /// initialization is the Initialize method.
        /// </summary>
        public VsExt_CopyFileNamePackage()
        {
            Trace.WriteLine(string.Format(CultureInfo.CurrentCulture, "Entering constructor for: {0}", this.ToString()));
        }

        /// <summary>
        /// Gets or sets the DTE.
        /// </summary>
        /// <value>The DTE.</value>
        private DTE2 Dte { get; set; }

        /// <summary>
        /// Gets or sets the solution explorer.
        /// </summary>
        /// <value>The solution explorer.</value>
        private UIHierarchy SolutionExplorer { get; set; }

        /// <summary>
        /// Initialization of the package; this method is called right after the package is sited, so this is the place
        /// where you can put all the initialization code that rely on services provided by VisualStudio.
        /// </summary>
        protected override void Initialize()
        {
            Trace.WriteLine(string.Format(CultureInfo.CurrentCulture, "Entering Initialize() of: {0}", this.ToString()));

            base.Initialize();

            Dte = (DTE2)GetService(typeof(DTE));

            // Add our command handlers for menu (commands must exist in the .vsct file)
            OleMenuCommandService mcs = GetService(typeof(IMenuCommandService)) as OleMenuCommandService;

            if (null != mcs)
            {
                // Create the command for the menu item.
                CommandID menuCommandID = new CommandID(GuidList.guidVsExt_CopyFileNameCmdSet, (int)PkgCmdIDList.cmdidMyCommand);
                MenuCommand menuItem = new MenuCommand(MenuItemCallback, menuCommandID);
                mcs.AddCommand(menuItem);
            }
        }

        /// <summary>
        /// Gets the selected UI hierarchy.
        /// </summary>
        /// <returns></returns>
        private ProjectItem GetSelectedUIHierarchy(UIHierarchy solutionExplorer)
        {
            ProjectItem item = null;

            object[] selection = solutionExplorer.SelectedItems as object[];

            if (selection.Length != 1)
                return item;

            if ((selection[0] as UIHierarchyItem).Object is ProjectItem)
                item = (ProjectItem)(selection[0] as UIHierarchyItem).Object;

            return item;
        }

        /// <summary>
        /// Determines whether [is command executable].
        /// </summary>
        /// <returns>
        /// 	<c>true</c> if [is command executable]; otherwise, <c>false</c>.
        /// </returns>
        private bool IsCommandExecutable()
        {
            bool executable = false;

            try
            {
                // Get the the Solution Explorer tree
                SolutionExplorer = Dte.ToolWindows.SolutionExplorer;

                // Check if there is any open solution
                if ((SolutionExplorer.UIHierarchyItems.Count > 0))
                {
                    executable = true;
                }
            }
            catch (Exception)
            {
                executable = false;
            }

            return executable;
        }

        /// <summary>
        /// This function is the callback used to execute a command when the a menu item is clicked.
        /// See the Initialize method to see how the menu item is associated to this function using
        /// the OleMenuCommandService service and the MenuCommand class.
        /// </summary>
        private void MenuItemCallback(object sender, EventArgs e)
        {
            if (!IsCommandExecutable())
                return;

            ProjectItem item = GetSelectedUIHierarchy(SolutionExplorer);

            if (item == null)
                return;

            Clipboard.SetText(item.Name);
        }
    }
}
